package mine.code.question.动态规划;

import org.junit.Test;

/**
 * 超级丑数 是一个正整数，并满足其所有质因数都出现在质数数组 primes 中。
 * <p>
 * 给你一个整数 n 和一个整数数组 primes ，返回第 n 个 超级丑数 。
 * <p>
 * 题目数据保证第 n 个 超级丑数 在 32-bit 带符号整数范围内。
 * <p>
 * 示例 1：
 * <p>
 * 输入：n = 12, primes = [2,7,13,19]
 * 输出：32
 * 解释：给定长度为 4 的质数数组 primes = [2,7,13,19]，前 12 个超级丑数序列为：[1,2,4,7,8,13,14,16,19,26,28,32] 。
 *
 * @author caijinnan
 */
public class 超级丑数 {

    @Test
    public void run() {
        int n = 12;
        int[] primes = {2, 7, 13, 19};
        System.out.println(nthSuperUglyNumber(n, primes));
    }

    public int nthSuperUglyNumber(int n, int[] primes) {
        int[] dp = new int[n];
        dp[0] = 1;
        int[] index = new int[primes.length];
        for (int i = 1; i < n; i++) {
            int tempMin = Integer.MAX_VALUE;
            for (int j = 0; j < primes.length; j++) {
                int num = primes[j] * dp[index[j]];
                if (num < tempMin) {
                    tempMin = num;
                }
            }
            for (int j = 0; j < primes.length; j++) {
                int num = primes[j] * dp[index[j]];
                if (num == tempMin) {
                    index[j] = index[j] + 1;
                }
            }
            dp[i] = tempMin;
        }
        return dp[n - 1];
    }
}
